import type { Router, RouteLocationNormalizedGeneric } from 'vue-router'
import NProgress from 'nprogress'

import { useTitle } from '@/hooks/useTitle'
import { setRouteChange } from '@/hooks/useRouteListener'
import { isWhiteList } from './whitelist'
import { getToken } from '@/utils/cache/cookies'
import { useUserStore } from '@/pinia/stores/user'
import { usePermissionStore } from '@/pinia/stores/permission'
import { routerConfig } from '@/router/config'

NProgress.configure({ showSpinner: false }) // NProgress Configuration

const { setTitle } = useTitle()

const LOGIN_PATH = '/login'

export function registerNavigationGuard(router: Router) {
  // 全局前置路由守卫
  router.beforeEach(async (to, from) => {
    // start progress bar
    NProgress.start()
    const userStore = useUserStore()
    const permissionStore = usePermissionStore()
    // 如果没有登录
    if (!getToken()) {
      // 如果在免登录的白名单中，则直接进入
      if (isWhiteList(to)) return true
      // 其他没有访问权限的页面将被重定向到登录页面
      return { path: LOGIN_PATH, query: { redirect: to.path } }
    }

    // 如果已经登录，并准备进入 Login 页面，则重定向到主页
    if (to.path === LOGIN_PATH) return '/'
    // 如果用户已经获得其权限角色
    if (userStore.roles.length !== 0) return true
    // 否则要重新获取权限角色
    try {
      await userStore.getInfo()
      // 注意：角色必须是一个数组！ 例如: ["admin"] 或 ["developer", "editor"]
      const roles = userStore.roles
      // 生成可访问的 Routes
      if (routerConfig.dynamic) permissionStore.setRoutes(roles)
      else permissionStore.setAllRoutes()
      // 将 "有访问权限的动态路由" 添加到 Router 中
      permissionStore.addRoutes.forEach((route) => router.addRoute(route))
      // 设置 replace: true, 因此导航将不会留下历史记录
      return { ...to, replace: true }
    } catch (error) {
      // 过程中发生任何错误，都直接重置 Token，并重定向到登录页面
      userStore.resetToken()
      ElMessage.error((error as Error).message || '路由守卫发生错误')
      return LOGIN_PATH
    }
  })

  // 全局后置钩子
  router.afterEach((to: RouteLocationNormalizedGeneric) => {
    setRouteChange(to)
    setTitle(to.meta.title)
    NProgress.done()
  })
}
